package com.gph.saviorframework.security.service.impl;

import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.gph.saviorframework.Constants;
import com.gph.saviorframework.FieldType;
import com.gph.saviorframework.common.model.Function;
import com.gph.saviorframework.common.model.Role;
import com.gph.saviorframework.dao.security.FunctionDao;
import com.gph.saviorframework.dao.security.RoleDao;
import com.gph.saviorframework.security.service.RoleService;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

public class RoleServiceImpl implements RoleService {

    @Autowired
    private RoleDao roleDao;

    @Autowired
    private FunctionDao functionDao;

    private Map<String, String> fields = new HashMap<String, String>();

    public RoleServiceImpl() {
        fields.put("ROLE_NAME", FieldType.STRING);
        fields.put("ROLE_DESCRIPTION", FieldType.STRING);
    }

    /**
     * 保存
     *
     * @param role
     */
    public void save(Role role) {
        roleDao.save(role);
    }

    /**
     * 根据编号获取
     *
     * @param id
     * @return
     */
    public Role get(String id) {
        return roleDao.get(id);
    }

    /**
     * 更新
     *
     * @param role
     * @return
     */
    public void update(Role role) {
        roleDao.update(role);
    }

    /**
     * 批量更新
     *
     * @param roles
     */
    @Override
    public void update(Set<Role> roles) {
        for (Role role : roles) {
            update(role);
        }
    }

    /**
     * 根据条件返回数据
     *
     * @param sort
     * @param dir
     * @param params
     * @return
     */
    public List<Map<String, Object>> find(String sort, String dir, Map<String, Object> params) {
        return roleDao.find(sort, dir, genWhereClause(params));
    }

    /**
     * 带分页参数的，给bootstrap使用的
     *
     * @param start  开始页码，从1开始
     * @param limit  每页查询的次数
     * @param sort
     * @param dir
     * @param params
     * @return
     */
    @Override
    public Map<String, Object> find(int start, int limit, String sort, String dir, Map<String, Object> params, Object... extras) {
        Map<String, Object> map = new HashMap<String, Object>();

        PageHelper.startPage(start, limit);
        List<Map<String, Object>> list = find(sort, dir, params);
        PageInfo pageInfo = new PageInfo(list);

        map.put(Constants.DEFAULT_RECORD_MODEL_KEY, list);
        map.put(Constants.DEFAULT_COUNT_MODEL_KEY, pageInfo.getPages());
        map.put(Constants.DEFAULT_RECORDS_MODEL_KEY, pageInfo.getTotal());
        map.put(Constants.DEFAULT_SUCCESS_KEY, Boolean.TRUE);

        return map;
    }

    /**
     * 统计数量
     *
     * @param params
     * @return
     */
    public long count(Map<String, Object> params) {
        return roleDao.count(genWhereClause(params));
    }

    /**
     * 根据主键删除
     *
     * @param id
     */
    public void delete(String id) {
        roleDao.deleteById(id);
    }

    /**
     * 批量删除
     *
     * @param ids
     */
    public void deleteByIds(String[] ids) {
        for (String id : ids) {
            delete(id);
        }
    }

    /**
     * 查询字段集合
     *
     * @return
     */
    public Map<String, String> getFields() {
        return fields;
    }

    /**
     * 根据id查询map
     *
     * @param id
     * @return
     */
    public Map<String, Object> findById(String id) {
        return roleDao.findById(id);
    }

    /**
     * 生成where条件
     *
     * @param params
     * @param extra
     * @return
     */
    public String genWhereClause(Map<String, Object> params, Object... extra) {
        String whereClause = "1=1";
        if (params != null && params.keySet().size() > 0) {
            for (String key : params.keySet()) {
                if (fields.get(key).equals(FieldType.STRING)) {
                    whereClause += " AND R." + key + " LIKE '" + params.get(key) + "'";
                } else {
                    whereClause += " AND R." + key + " ='" + params.get(key) + "'";
                }
            }
        }
        return whereClause;
    }

    /**
     * 保存role-function之间的关系,新建用户具有index的访问权限
     *
     * @param role
     */
    public void saveWithFunction(Role role) {
        roleDao.save(role);
        //保存role-function之间的关系
        Function index = functionDao.get("INDEX");// 新建角色默认具有首页访问权限
        if (index != null) {
            addRoleRelFunction(role.getId(), index.getId());
        }
    }

    /**
     * 保存角色和菜单之间的关系
     *
     * @param roleId
     * @param menuId
     */
    public void addRoleRelMenu(String roleId, String menuId) {
        roleDao.addRoleRelMenu(roleId, menuId);
    }

    /**
     * 保存角色和菜单编号之间的关系
     *
     * @param menuIds
     * @param roleId
     */
    public void saveRoleMenu(String[] menuIds, String roleId) {
        //删除原先的关系
        roleDao.deleteRoleRelMenu(roleId);
        if (menuIds == null) {
            return;
        }
        for (String funId : menuIds) {
            addRoleRelMenu(roleId, funId);
        }
    }

    /**
     * 保存角色和功能之间的关系
     *
     * @param funIds
     * @param roleId
     */
    public void saveRoleFunction(String[] funIds, String roleId) {
        //删除原先的关系
        roleDao.deleteRoleRelFunction(roleId);
        if (funIds == null) {
            return;
        }
        for (String funId : funIds) {
            addRoleRelFunction(roleId, funId);
        }
    }

    /**
     * 添加角色和功能之间的关系
     *
     * @param roleId
     * @param funId
     */
    public void addRoleRelFunction(String roleId, String funId) {
        roleDao.addRoleRelFunction(roleId, funId);
    }

    /**
     * 保存用户和角色之间的关系
     *
     * @param checked
     * @param unchecked
     * @param roleId
     */
    public void saveRoleUser(String[] checked, String[] unchecked, String roleId) {
        roleDao.deleteRoleRelUser(roleId, "");
        for (String username : checked) {
            roleDao.addRoleRelUser(roleId, username);
        }
    }

    /**
     * 根据用户编号获取记录
     *
     * @param user
     * @return
     */
    public List<Map<String, Object>> findByUser(String user) {
        return roleDao.findByUser(user);
    }

    /**
     * 根据岗位获取角色
     *
     * @param positionId
     * @return
     */
    public List<Map<String, Object>> findByPosition(String positionId) {
        return roleDao.findByPosition(positionId);
    }
}
